Skip to content

Fix JAX version bugs for nightly image #2503

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Closed
wants to merge 5 commits into from
Closed

Fix JAX version bugs for nightly image #2503

wants to merge 5 commits into from

Conversation

@hengtaoguo
Copy link
Collaborator

@hengtaoguo hengtaoguo commented Oct 15, 2025 •
edited
Loading

Description

Skip the JAX 0.7.0 installation for the nightly images. Only do so for the stable_stack used in GitHub CI.

b/451429959

Tests

N/A

Checklist

Before submitting this PR, please make sure (put X in square brackets):

  • I have performed a self-review of my code. For an optional AI review, add the gemini-review label.
  • I have necessary comments in my code, particularly in hard-to-understand areas.
  • I have run end-to-end tests tests and provided workload links above if applicable.
  • I have made or will make corresponding changes to the doc if needed, including adding new documentation pages to the relevant Table of Contents (toctree directive) as explained in our documentation.

@gobbleturk
Copy link
Collaborator

gobbleturk commented Oct 15, 2025

This may or may not fix the actual issue; transitive dependencies like tunix might still force a different jax version than we desire.

Regardless this PR looks like a strict improvement, thanks!

@github-actions
Copy link

github-actions bot commented Nov 16, 2025

This PR has been automatically marked as stale because it has not had recent activity. It will be closed soon if no further activity occurs. Thank you for your contributions.

@github-actions github-actions bot added the stale Automatically applied to stale PRs. label Nov 16, 2025
@github-actions
Copy link

github-actions bot commented Nov 23, 2025

This PR was closed because it has been inactive for a while. Please reopen it if you are still working on it.

@github-actions github-actions bot closed this Nov 23, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@bvandermoon bvandermoon Awaiting requested review from bvandermoon bvandermoon will be requested when the pull request is marked ready for review bvandermoon is a code owner

@parambole parambole Awaiting requested review from parambole parambole will be requested when the pull request is marked ready for review parambole is a code owner

@richjames0 richjames0 Awaiting requested review from richjames0 richjames0 will be requested when the pull request is marked ready for review richjames0 is a code owner

@shralex shralex Awaiting requested review from shralex shralex will be requested when the pull request is marked ready for review shralex is a code owner

Assignees

@hengtaoguo hengtaoguo

@gobbleturk gobbleturk

Labels

stale Automatically applied to stale PRs.

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@hengtaoguo @gobbleturk